package 搜索.广度优先搜索.middle;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.Queue;
import java.util.Set;

/**
 * @author Lw中
 * @date 2020/10/18 11:50
 */
public class ErQiJiu {

    /**
     * LWZ TODO : 2020/10/18 没有看懂
     *
     */

    /**
     * 完全平方数
     * 使用广度优先搜索算法
     * @param n
     */
    public int numSquares(int n) {
        Queue<Integer> queue = new LinkedList<>();
        /* 用于存储访问过的元素 */
        Set<Integer> visited = new HashSet<>();
        queue.add(0);
        visited.add(0);
        /* 用于存储平方数的个数 */
        int distance = 0;
        while (!queue.isEmpty()) {
            distance++;
            int size = queue.size();
            for (int i = 0; i < size; i++) {
                /* 获取对头元素 */
                int curr = queue.poll();
                for (int j = 1; j * j + curr <= n; j++) {
                    int next = j * j + curr;
                    if (next == n) {
                        return distance;
                    }
                    if (next < n && !visited.contains(next)) {
                        queue.add(next);
                        visited.add(next);
                    }
                }
            }
        }
        return distance;
    }

    public static void main(String[] args) {

    }

}
